Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,7 +72,6 @@ def load_model(model, path=r"C:\Users\nandu\Documents\travis\models\gpt_model.pt
|
|
| 72 |
else:
|
| 73 |
print("Model file not found!")
|
| 74 |
|
| 75 |
-
load_model(model)
|
| 76 |
|
| 77 |
def generate_response_stream(model, query, max_length=200):
|
| 78 |
model.eval()
|
|
@@ -123,4 +122,5 @@ def query_model():
|
|
| 123 |
return Response(stream_with_context(generate()), mimetype='text/event-stream')
|
| 124 |
|
| 125 |
if __name__ == "__main__":
|
|
|
|
| 126 |
app.run(host="0.0.0.0", port=7860)
|
|
|
|
| 72 |
else:
|
| 73 |
print("Model file not found!")
|
| 74 |
|
|
|
|
| 75 |
|
| 76 |
def generate_response_stream(model, query, max_length=200):
|
| 77 |
model.eval()
|
|
|
|
| 122 |
return Response(stream_with_context(generate()), mimetype='text/event-stream')
|
| 123 |
|
| 124 |
if __name__ == "__main__":
|
| 125 |
+
load_model(model)
|
| 126 |
app.run(host="0.0.0.0", port=7860)
|