Commit 
							
							·
						
						6a4f7b3
	
1
								Parent(s):
							
							869aa0f
								
Added missing quotes for tokenizer
Browse files
    	
        app.py
    CHANGED
    
    | @@ -2,16 +2,16 @@ from transformers import AutoModelForCausalLM, AutoTokenizer | |
| 2 | 
             
            import gradio as gr
         | 
| 3 |  | 
| 4 | 
             
            model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instruct")
         | 
| 5 | 
            -
            tokenizer = AutoTokenizer.from_pretrained(deepseek- | 
| 6 |  | 
| 7 | 
            -
            def generate_code(prompt)
         | 
| 8 | 
            -
                inputs = tokenizer(prompt, return_tensors=pt)
         | 
| 9 | 
            -
                outputs = model.generate(inputs, max_length=200)
         | 
| 10 | 
             
                return tokenizer.decode(outputs[0], skip_special_tokens=True)
         | 
| 11 |  | 
| 12 | 
             
            gr.Interface(
         | 
| 13 | 
             
                fn=generate_code,
         | 
| 14 | 
            -
                inputs=text,
         | 
| 15 | 
            -
                outputs=text,
         | 
| 16 | 
            -
                title=DeepSeek Coder 1.3B
         | 
| 17 | 
             
            ).launch()
         | 
|  | |
| 2 | 
             
            import gradio as gr
         | 
| 3 |  | 
| 4 | 
             
            model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instruct")
         | 
| 5 | 
            +
            tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instruct")
         | 
| 6 |  | 
| 7 | 
            +
            def generate_code(prompt):
         | 
| 8 | 
            +
                inputs = tokenizer(prompt, return_tensors="pt")
         | 
| 9 | 
            +
                outputs = model.generate(**inputs, max_length=200)
         | 
| 10 | 
             
                return tokenizer.decode(outputs[0], skip_special_tokens=True)
         | 
| 11 |  | 
| 12 | 
             
            gr.Interface(
         | 
| 13 | 
             
                fn=generate_code,
         | 
| 14 | 
            +
                inputs="text",
         | 
| 15 | 
            +
                outputs="text",
         | 
| 16 | 
            +
                title="DeepSeek Coder 1.3B"
         | 
| 17 | 
             
            ).launch()
         | 
