Update main.py

main.py

@@ -1,4 +1,8 @@
-from app import app
+from load_model import load_model
 
-if __name__ == '__main__':
-    app.run(debug=True)
+model, tokenizer = load_model()
+
+def generate_response(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(inputs['input_ids'])
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)