Update main.py

main.py

@@ -8,13 +8,13 @@ import gradio as gr
 
 #app = FastAPI()
 
-pipe_flan = pipeline("text-generation", model="gpt2")
+#pipe_flan = pipeline("text-generation", model="gpt2")
 
 #@app.get("/infer_gpt2")
-def gpt2(input):
-    output = pipe_flan(input, max_length=130, num_return_sequences=1)
+#def gpt2(input):
+#    output = pipe_flan(input, max_length=130, num_return_sequences=1)
     #return {"output": output[0]["generated_text"]}
-    return {"output": output[0]["generated_text"]}
+#    return {"output": output[0]["generated_text"]}
 
 #app.mount("/", StaticFiles(directory="static", html=True), name="static")