Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,9 +55,7 @@ def query_local_model_transformers(text):
|
|
| 55 |
pred = pipe(text, max_length=1000)
|
| 56 |
print(type(pred))
|
| 57 |
print(pred)
|
| 58 |
-
return pred
|
| 59 |
-
#pred = llama(text, temperature=0.1, max_tokens=500)
|
| 60 |
-
#return pred["choices"][0]["text"]
|
| 61 |
|
| 62 |
|
| 63 |
def query_local_model(text):
|
|
|
|
| 55 |
pred = pipe(text, max_length=1000)
|
| 56 |
print(type(pred))
|
| 57 |
print(pred)
|
| 58 |
+
return pred[0]["generated_text"]
|
|
|
|
|
|
|
| 59 |
|
| 60 |
|
| 61 |
def query_local_model(text):
|