Update app.py

app.py

@@ -55,9 +55,7 @@ def query_local_model_transformers(text):
     pred = pipe(text, max_length=1000)
     print(type(pred))
     print(pred)
-    return pred
-    #pred = llama(text, temperature=0.1, max_tokens=500)
-    #return pred["choices"][0]["text"]
+    return pred[0]["generated_text"]
 
 
 def query_local_model(text):