Spaces:

Marthee
/

mouse_click

Runtime error

Marthee commited on Jun 14, 2023

Commit

a792e4a

1 Parent(s): 36741e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,17 +27,20 @@ def mouse_callback(event, x, y, flags, param):
 # Create a Flask app
 app = Flask(__name__)
 # Define the Gradio interface
-iface = gr.Interface(fn=None, inputs="image", outputs=None)
 # Define the HTML template
 html_template = '''
 <!DOCTYPE html>
 <html>
 <body>
-    <h1>Upload an Image</h1>
     <form method="POST" action="/" enctype="multipart/form-data">
-        <input type="file" name="image">
         <input type="submit" value="Submit">
     </form>
     <p>{{ mouse_event }}</p>
@@ -49,17 +52,14 @@ html_template = '''
 @app.route('/', methods=['GET', 'POST'])
 def home():
     if request.method == 'POST':
-        # Get the uploaded image
-        image = cv2.imdecode(np.fromstring(request.files['image'].read(), np.uint8), cv2.IMREAD_COLOR)
-        # Display the image
-        cv2.imshow("Image", image)
-        cv2.setMouseCallback("Image", mouse_callback)
-        cv2.waitKey(0)
-        cv2.destroyAllWindows()
-        # Render the template with the mouse event
-        return render_template_string(html_template, mouse_event=app.mouse_event)
     # Render the template initially
     return render_template_string(html_template)

 # Create a Flask app
 app = Flask(__name__)
+# Load the Hugging Face pipeline for text generation
+text_generator = pipeline("text-generation", model="gpt2")
 # Define the Gradio interface
+iface = gr.Interface(fn=text_generator, inputs="text", outputs="text")
 # Define the HTML template
 html_template = '''
 <!DOCTYPE html>
 <html>
 <body>
+    <h1>Enter Text</h1>
     <form method="POST" action="/" enctype="multipart/form-data">
+        <input type="text" name="text">
         <input type="submit" value="Submit">
     </form>
     <p>{{ mouse_event }}</p>
 @app.route('/', methods=['GET', 'POST'])
 def home():
     if request.method == 'POST':
+        # Get the user input text
+        input_text = request.form['text']
+        # Generate text using Hugging Face pipeline
+        generated_text = text_generator(input_text, max_length=50)[0]['generated_text']
+        # Render the template with the generated text and mouse event
+        return render_template_string(html_template, generated_text=generated_text, mouse_event=app.mouse_event)
     # Render the template initially
     return render_template_string(html_template)