Spaces:
Runtime error
Runtime error
Commit
·
fd16522
1
Parent(s):
94dac17
Update app.py
Browse files
app.py
CHANGED
@@ -48,15 +48,15 @@ def the_process(input_text, model_choice):
|
|
48 |
pythonFlag = "true"
|
49 |
elif(pythonFlag == "true"):
|
50 |
print("pythonFlag", pythonFlag)
|
51 |
-
|
52 |
-
|
53 |
print("Inside llama for python")
|
54 |
tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
|
55 |
model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
|
56 |
output = run_predict(input_text, model, tokenizer)
|
57 |
print("output" , output)
|
58 |
javaFlag = "true"
|
59 |
-
|
60 |
print("javaFlag", javaFlag)
|
61 |
else:
|
62 |
a_variable = model_box[model_choice]
|
|
|
48 |
pythonFlag = "true"
|
49 |
elif(pythonFlag == "true"):
|
50 |
print("pythonFlag", pythonFlag)
|
51 |
+
if(model_choice==0):
|
52 |
+
if(javaFlag == "false"):
|
53 |
print("Inside llama for python")
|
54 |
tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
|
55 |
model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
|
56 |
output = run_predict(input_text, model, tokenizer)
|
57 |
print("output" , output)
|
58 |
javaFlag = "true"
|
59 |
+
elif(javaFlag == "true"):
|
60 |
print("javaFlag", javaFlag)
|
61 |
else:
|
62 |
a_variable = model_box[model_choice]
|