Spaces:
Runtime error
Runtime error
Commit
·
b47d696
1
Parent(s):
fe7f36f
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def the_process(input_text, model_choice):
|
|
45 |
model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf_python")
|
46 |
output = run_predict(input_text, model, tokenizer)
|
47 |
print("output starcoder python" , output)
|
48 |
-
|
49 |
if(javaFlag == "false"):
|
50 |
print("Inside starcoder for java")
|
51 |
tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
|
|
|
45 |
model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf_python")
|
46 |
output = run_predict(input_text, model, tokenizer)
|
47 |
print("output starcoder python" , output)
|
48 |
+
elif(model_choice==0):
|
49 |
if(javaFlag == "false"):
|
50 |
print("Inside starcoder for java")
|
51 |
tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
|