Update app.py

app.py

@@ -1,3 +1,26 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
-gr.load("models/cris177/Qwen2-Simple-Arguments").launch()
+model = AutoModelForCausalLM.from_pretrained("cris177/Qwen2-Simple-Arguments")
+tokenizer = AutoTokenizer.from_pretrained("cris177/Qwen2-Simple-Arguments")
+
+argument = "If it's wednesday it's cold, and it's cold, therefore it's wednesday."
+def analyze_argument(argument):
+    instruction = 'Based on the following argument, identify the following elements: premises, conclusion, propositions, type of argument, negation of propositions and validity.'
+    alpaca_prompt = """Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+
+    ### Instruction:
+    {}
+
+    ### Input:
+    {}
+
+    ### Response:"""
+    prompt = alpaca_prompt.format(instruction, argument)
+    input_ids = tokenizer(prompt, return_tensors="pt")
+
+    outputs = model.generate(**input_ids, max_length=1000, num_return_sequences=1)
+    return tokenizer.decode(outputs[0])
+
+
+gr.Interface(analyze_argument, inputs="text", outputs="text").launch()