Spaces:

Jaehan
/

Code-Generator-1

Runtime error

Jaehan commited on Jun 25, 2023

Commit

d3c2d2f

•

1 Parent(s): 626e0b9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 model_name = "Salesforce/codegen-350M-mono"
 codegen_token = AutoTokenizer.from_pretrained(model_name)
-model = AutoTokenizer.from_pretrained(model_name)
 def codegen(intent):
     """Give input as text which reflects intent of the program.

 model_name = "Salesforce/codegen-350M-mono"
 codegen_token = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 def codegen(intent):
     """Give input as text which reflects intent of the program.